package com.lsec.springbootinit.mybatis.mapper;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
import com.baomidou.mybatisplus.extension.toolkit.Db;
import com.lsec.springbootinit.model.common.PageParam;
import com.lsec.springbootinit.model.common.PageResult;
import com.lsec.springbootinit.utils.PageUtils;
import org.apache.ibatis.annotations.Param;

import java.util.Collection;
import java.util.List;


/**
 * 在 MyBatis Plus 的 BaseMapper 的基础上拓展，提供更多的能力
 * join多表还是编写xml  如果使用mybatis-plus-join 难维护,难阅读
 *
 * @author Cherry
 * @date 2023/06/02
 */
public interface BaseMapperX<T> extends BaseMapper<T> {

    /**
     * 查询分页列表
     *
     * @param pageParam    页面参数
     * @param queryWrapper 查询包装
     * @return {@link PageResult}<{@link T}>
     */
    default PageResult<T> selectPage(PageParam pageParam, @Param("ew") Wrapper<T> queryWrapper) {
        // MyBatis Plus 查询
        IPage<T> mpPage = PageUtils.buildPage(pageParam);
        selectPage(mpPage, queryWrapper);
        // 转换返回
        return PageUtils.convertToPageResult(mpPage);
    }

    /**
     * 查询单个
     *
     * @param field 场
     * @param value 价值
     * @return {@link T}
     */
    default T selectOne(String field, Object value) {
        return selectOne(new QueryWrapper<T>().eq(field, value));
    }

    /**
     * 查询单个
     *
     * @param field 场
     * @param value 价值
     * @return {@link T}
     */
    default T selectOne(SFunction<T, ?> field, Object value) {
        return selectOne(new LambdaQueryWrapper<T>().eq(field, value));
    }

    /**
     * 查询单个
     *
     * @param field1 field1
     * @param value1 value1
     * @param field2 field2
     * @param value2 value2
     * @return {@link T}
     */
    default T selectOne(String field1, Object value1, String field2, Object value2) {
        return selectOne(new QueryWrapper<T>().eq(field1, value1).eq(field2, value2));
    }

    /**
     * 查询单个
     *
     * @param field1 field1
     * @param value1 value1
     * @param field2 field2
     * @param value2 value2
     * @return {@link T}
     */
    default T selectOne(SFunction<T, ?> field1, Object value1, SFunction<T, ?> field2, Object value2) {
        return selectOne(new LambdaQueryWrapper<T>().eq(field1, value1).eq(field2, value2));
    }

    /**
     * 查询总数
     *
     * @return {@link Long}
     */
    default Long selectCount() {
        return selectCount(new QueryWrapper<T>());
    }

    /**
     * 查询总数
     *
     * @param field 场
     * @param value 价值
     * @return {@link Long}
     */
    default Long selectCount(String field, Object value) {
        return selectCount(new QueryWrapper<T>().eq(field, value));
    }

    /**
     * 查询总数
     *
     * @param field 场
     * @param value 价值
     * @return {@link Long}
     */
    default Long selectCount(SFunction<T, ?> field, Object value) {
        return selectCount(new LambdaQueryWrapper<T>().eq(field, value));
    }

    /**
     * 查询列表
     *
     * @return {@link List}<{@link T}>
     */
    default List<T> selectList() {
        return selectList(new QueryWrapper<>());
    }

    /**
     * 查询列表
     *
     * @param field 场
     * @param value 价值
     * @return {@link List}<{@link T}>
     */
    default List<T> selectList(String field, Object value) {
        return selectList(new QueryWrapper<T>().eq(field, value));
    }

    /**
     * 查询列表
     *
     * @param field 场
     * @param value 价值
     * @return {@link List}<{@link T}>
     */
    default List<T> selectList(SFunction<T, ?> field, Object value) {
        return selectList(new LambdaQueryWrapper<T>().eq(field, value));
    }

    /**
     * 查询列表
     *
     * @param field  场
     * @param values 值
     * @return {@link List}<{@link T}>
     */
    default List<T> selectList(String field, Collection<?> values) {
        return selectList(new QueryWrapper<T>().in(field, values));
    }

    /**
     * 查询列表
     *
     * @param field  场
     * @param values 值
     * @return {@link List}<{@link T}>
     */
    default List<T> selectList(SFunction<T, ?> field, Collection<?> values) {
        return selectList(new LambdaQueryWrapper<T>().in(field, values));
    }

    /**
     * 查询列表
     *
     * @param leField 乐场
     * @param geField 通用电气领域
     * @param value   价值
     * @return {@link List}<{@link T}>
     */
    default List<T> selectList(SFunction<T, ?> leField, SFunction<T, ?> geField, Object value) {
        return selectList(new LambdaQueryWrapper<T>().le(leField, value).ge(geField, value));
    }

    /**
     * 批量插入，适合大量数据插入
     *
     * @param entities 实体们
     */
    default void insertBatch(Collection<T> entities) {
        Db.saveBatch(entities);
    }

    /**
     * 批量插入，适合大量数据插入
     *
     * @param entities 实体们
     * @param size     插入数量 Db.saveBatch 默认为 1000
     */
    default void insertBatch(Collection<T> entities, int size) {
        Db.saveBatch(entities, size);
    }

    /**
     * 批处理更新
     *
     * @param update 更新
     */
    default void updateBatch(T update) {
        update(update, new QueryWrapper<>());
    }

    /**
     * 批处理更新
     *
     * @param entities 实体
     * @param size     大小
     */
    default void updateBatch(Collection<T> entities, int size) {
        Db.updateBatchById(entities, size);
    }

}
